from ssd.modeling import registry
from .vgg import VGG
from .mobilenet import MobileNetV2
from .efficient_net import EfficientNet
from .mobilenetv3 import MobileNetV3

__all__ = ['build_backbone', 'VGG', 'MobileNetV2', 'EfficientNet', 'MobileNetV3']


def build_backbone(cfg):
    '''
    创建ssd 骨干网络
    骨干网路，主要使用图片分类神经网路实现图片的特征提取，可以使用常见的目标分类检测网络
    '''
    # todo 看到这里
    return registry.BACKBONES[cfg.MODEL.BACKBONE.NAME](cfg, cfg.MODEL.BACKBONE.PRETRAINED)
